import pandas as pd
import time
from datetime import timedelta

def load_csv_basic(csv_path):
    start_time = time.time()  # 记录开始时间
    
    # 直接加载（适合内存充足情况）
    df = pd.read_csv(
        csv_path,
        names=['user_id', 'item_id', 'category_id', 'behavior_type', 'timestamp'],
        header=None,
        parse_dates=False,
        infer_datetime_format=False,
        low_memory=False
    )
    
    end_time = time.time()  # 记录结束时间
    elapsed = end_time - start_time  # 计算耗时（秒）
    # 转换为人类可读格式（时:分:秒.毫秒）
    print(f"基础加载耗时: {str(timedelta(seconds=elapsed))}")
    return df

# 执行
csv_path = r'D:\data\公开数据集\淘宝用户行为数据\UserBehavior.csv'
df_view = pd.read_csv(csv_path, nrows=5, header=None)
df_basic = load_csv_basic(csv_path)
print(df_basic.head())